package com.stone.boot.stoneboot.utils.generator;

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.config.rules.DbColumnType;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import com.baomidou.mybatisplus.generator.fill.Column;

import java.sql.Types;
import java.util.*;

/**
 * mybatis Plus 的代码生成.
 */
public class MybatisPlusGenerator {
    public static void main(String[] args) {
        String strWindows = "C:\\Users\\lipw\\Downloads";
        generator(strWindows);
    }

    /**
     * 提前在代码中写入信息直接生成.
     */
    public static void generator(String outPutDir) {

        FastAutoGenerator.create("jdbc:mysql://192.168.99.167:3306/oa", "oa", "YMcpA4iyeaRfxZKm")
                .globalConfig(builder -> {
                    builder.author("lipw") // 设置作者
                            .enableSwagger() // 开启 swagger 模式
                            .outputDir(outPutDir); // 指定输出目录
                })
                .dataSourceConfig(builder ->
                        builder.typeConvertHandler((globalConfig, typeRegistry, metaInfo) -> {
                            int typeCode = metaInfo.getJdbcType().TYPE_CODE;
                            if (typeCode == Types.SMALLINT) {
                                // 自定义类型转换
                                return DbColumnType.INTEGER;
                            }
                            return typeRegistry.getColumnType(metaInfo);
                        })
                )
                .packageConfig(builder ->
                        builder.parent("com.bkht") // 设置父包名
                                .moduleName("oa") // 设置父包模块名
                                .pathInfo(Collections.singletonMap(OutputFile.xml, outPutDir)) // 设置mapperXml生成路径
                )
                .strategyConfig(builder ->
                                builder.addInclude("oa_admin") // 设置需要生成的表名
                                        .addTablePrefix("oa_", "c_") // 设置过滤表前缀
//                                .entityBuilder().javaTemplate("")
//                                .serviceBuilder().serviceTemplate("")
//                                .serviceImplTemplate("")
                                        .controllerBuilder().template("/templates/yd/controller.java")
//                                .mapperBuilder().mapperTemplate("").mapperXmlTemplate("")
                )
                .injectionConfig(consumer -> {
                    Map<String, String> customFile = new HashMap<>();
                    customFile.put("Data.sql", "/templates/yd/power.sql.ftl");
                    consumer.customFile(customFile);
                })
                .templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板，默认的是Velocity引擎模板
                .execute();
    }

    public static void generator2() {
        FastAutoGenerator.create("url", "username", "password")
                // 全局配置
                .globalConfig((scanner, builder) -> builder.author(scanner.apply("请输入作者名称？")))
                // 包配置
                .packageConfig((scanner, builder) -> builder.parent(scanner.apply("请输入包名？")))
                // 策略配置
                .strategyConfig((scanner, builder) -> builder.addInclude(getTables(scanner.apply("请输入表名，多个英文逗号分隔？所有输入 all")))
                        .entityBuilder()
                        .enableLombok()
                        .addTableFills(
                                new Column("create_time", FieldFill.INSERT)
                        )
                        .build())
                // 使用Freemarker引擎模板，默认的是Velocity引擎模板
                .templateEngine(new FreemarkerTemplateEngine())
                .execute();
    }

    // 处理 all 情况
    protected static List<String> getTables(String tables) {
        return "all".equals(tables) ? Collections.emptyList() : Arrays.asList(tables.split(","));
    }
}
